python map function vs list comp